export function detectLanguage(code: string): string {
  const trimmed = code.trim();

  // Python 特征
  if (
    /(^|\n)\s*def\s+\w+\s*\(.*\)\s*:/.test(trimmed) || // def 函数定义
    /(^|\n)\s*import\s+\w+/.test(trimmed) ||          // import 模块
    /(^|\n)\s*from\s+\w+\s+import\s+/.test(trimmed) ||// from xxx import yyy
    /print\s*\(.*\)/.test(trimmed)                    // print 语句
  ) {
    return 'python';
  }

  // C++ 特征
  if (
    /#include\s*[<"]/.test(trimmed) ||       // include 头文件
    /using\s+namespace\s+\w+/.test(trimmed) ||         // using namespace
    /\bstd::\w+/.test(trimmed) ||                      // std::cout 等
    /\b(cout\s*<<|cin\s*>>)/.test(trimmed) ||           // cout, cin
    /\bclass\s+\w+\s*\{/.test(trimmed) ||               // class {
    /->/.test(trimmed) ||                              // a->b
    /::/.test(trimmed)                                 // A::B
  ) {
    return 'cpp';
  }

  // Java 特征
  if (
    /\b(public|private|protected)\s+(static\s+)?(void|String|int|long|double|float)\s+\w+\s*\(/.test(trimmed) ||
    /\bpublic\s+class\b/.test(trimmed) ||              // class 声明
    /import\s+java\./.test(trimmed) ||                 // import java.*
    /\bpublic\s+static\s+void\s+main\b/.test(trimmed) ||// main 方法
    /\bSystem\.out\.println\s*\(/.test(trimmed) ||       // 打印语句
    /@(Override|Test|Data)/.test(trimmed) ||           // 注解
    /\bnew\s+\w+\(/.test(trimmed)                      // new 关键字
  ) {
    return 'java';
  }

  // JavaScript 特征
  if (
    /\bfunction\s+\w+\s*\(/.test(trimmed) ||           // function 函数定义
    /\bconsole\.log\s*\(/.test(trimmed) ||             // 打印
    /\bvar\s+|let\s+|const\s+/.test(trimmed)           // 变量声明
  ) {
    return 'javascript';
  }

  // JSON 特征
  if (
    /^[\s\r\n]*\{[\s\S]*\}[\s\r\n]*$/.test(trimmed) &&  // 以 {} 包围
    /^[\s\r\n]*"[^"]+"\s*:\s*/.test(trimmed)           // 键值对结构
  ) {
    return 'json';
  }

  // Shell/bash 特征
  if (
    /^#!/.test(trimmed) ||                             // shebang
    /\becho\b/.test(trimmed) ||                        // echo
    /\bcd\b|\bexport\b/.test(trimmed)                  // 常见命令
  ) {
    return 'bash';
  }

  // 默认返回 plain text
  return 'plaintext';
}
